import os #进入OS模块

def size(dir):
    b = 0
    a = os.listdir(dir)#列出目录名下的所有文件
    for i in a :
        file1 = os.path.join(dir,i)
        if os.path.isfile(file1):#判断是不是文件
            c = os.path.getsize(file1)
            b += c
        if os.path.isdir(file1):#判断是不是目录
            b += size(file1)
    return b
#测试
b = size("./aa")
print(b)
